import pickle


def read_obj_pose(pose_path):
    """
    input: pose file path
    return:
    pose = {
        "rotation": rotation matrix, shape = (3, 3), dtype = np.float32,
        "translation": {position_x, position_y, position_z} (unit: m), shape = (3, 1), dtype = np.float32,
        (optional) "scale": shape = {length_x, length_y, length_z} (unit: m), (3, 1), dtype = np.float32,
    }
    """

    pose = pickle.load(open(pose_path, "rb"))
    return pose


if __name__ == "__main__":
    init_pose_path = ""  # TODO
    pose = read_obj_pose(init_pose_path)
